from torch_geometric.datasets import Planetoid
from torch_geometric.transforms import NormalizeFeatures

dataset = Planetoid(root = "D:/code/dataset/Planetoid",name="Cora",transform=NormalizeFeatures())

print(dataset)
print("图的个数",len(dataset))
print("每个节点的特征维度",dataset.num_features)
print("分类数",dataset.num_classes)

data = dataset[0]
# 2708个节点 连线为 10556/2
# Data(x=[2708, 1433], edge_index=[2, 10556], y=[2708], train_mask=[2708], val_mask=[2708], test_mask=[2708])
print(data)
